package com.sosky;

/**
 * 537. 复数乘法
 * <p>
 * 给定两个表示复数的字符串。
 * 返回表示它们乘积的字符串。注意，根据定义 i2 = -1 。
 * <p>
 * 示例 1:
 * 输入: "1+1i", "1+1i"
 * 输出: "0+2i"
 * 解释: (1 + i) * (1 + i) = 1 + i2 + 2 * i = 2i ，你需要将它转换为 0+2i 的形式。
 * <p>
 * 示例 2:
 * 输入: "1+-1i", "1+-1i"
 * 输出: "0+-2i"
 * 解释: (1 - i) * (1 - i) = 1 + i2 - 2 * i = -2i ，你需要将它转换为 0+-2i 的形式。
 * <p>
 * 注意:
 * 输入字符串不包含额外的空格。
 * 输入字符串将以 a+bi 的形式给出，其中整数 a 和 b 的范围均在 [-100, 100] 之间。输出也应当符合这种形式。
 */
public class Solution537 {
    public static void main(String[] args) {
        System.out.println(complexNumberMultiply("1+1i", "1+1i"));
        System.out.println(complexNumberMultiply("1+-1i", "1+-1i"));
    }

    public static String complexNumberMultiply(String num1, String num2) {
        String[] num1s = num1.split("\\+");
        String[] num2s = num2.split("\\+");
        int real = Integer.parseInt(num1s[0]) * Integer.parseInt(num2s[0]) + Integer.parseInt(num1s[1].replace("i",
                "")) * Integer.parseInt(num2s[1].replace("i",
                "")) * -1;
        int imaginary =
                Integer.parseInt(num1s[0]) * Integer.parseInt(num2s[1].replace("i", ""))
                        + Integer.parseInt(num1s[1].replace("i", "")) * Integer.parseInt(num2s[0]);
        return real + "+" + imaginary + "i";
    }
}
